from mmseg.datasets.builder import DATASETS
from mmseg.datasets.custom import CustomDataset


@DATASETS.register_module()
class Camvid12(CustomDataset):
    """ADE20K dataset.

    In segmentation map annotation for ADE20K, 0 stands for background, which
    is not included in 150 categories. ``reduce_zero_label`` is fixed to True.
    The ``img_suffix`` is fixed to '.jpg' and ``seg_map_suffix`` is fixed to
    '.png'.
    """
    CLASSES =('Bicyclist','Building','Car','Column_Pole','Fence','Pedestrian','Road'
        'Sidewalk','SignSymbol','Sky','Tree','backgroud')
    PALETTE = [[0, 128, 192],[128, 0, 0],[64, 0, 128],[192, 192, 128],[64, 64, 128],[64, 64, 0],[128, 64, 128],
       [0, 0, 192],[192, 128, 128], [128, 128, 128],[128, 128, 0],[0,0,0]]
    # [[0, 128, 192], [128, 0, 0], [64, 0, 128], [192, 192, 128], [64, 64, 128], [64, 64, 0], [128, 64, 128],
    # [0, 0, 192], [192, 128, 128], [128, 128, 128], [128, 128, 0], [0, 0, 0]]

    def __init__(self, **kwargs):
        super(Camvid12, self).__init__(
            img_suffix='.png',
            seg_map_suffix='_L.png',
            reduce_zero_label=True,
            **kwargs)
